/* libwatopt++ - Optimization Toolkit
 *
 * Copyright (c) 2007 Christopher Alexander Watford
 * <christopher.watford@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to
 * deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 *
 * $Id: art2_network.hpp 2 2007-05-06 18:01:15Z christopher.watford $
 */

#if _MSC_VER > 1000
#pragma once
#endif /* _MSC_VER > 1000 */

#ifndef _ART2_NETWORK_HPP_
#define _ART2_NETWORK_HPP_

#include <vector>
#include <iostream>

namespace watopt
{

   class ART2Network
   {
   private:
	   unsigned int		_f1_nodes;
	   unsigned int		_f2_nodes;
      unsigned int      _leader;

	   double				_aa,
						   _bb,
						   _cc,
						   _dd,
						   _rho,
						   _theta,
						   _eps;

	   double				*_ww, _ww_norm;
	   double				*_xx;
	   double				*_vv, _vv_norm;
	   double				*_uu, _uu_norm;
	   double				*_pp, _pp_norm;
	   double				*_qq;
      double            *_Tj;

	   double				*_rr, _rr_norm;

	   bool*				_reset;
	   bool				_f2_active;

	   double	*_zij;
      double	*_zji;
	   unsigned int		_JJ;
	   double	*_yy;

      void equilibrate(double* _II);
	   void find_max(void);
	   bool check_reset(void);
	   void apply_dz(void);

      void do_w(double* _II);
      void do_x(void);
      void do_v(void);
      void do_u(void);
      void do_p(void);
      void do_q(void);
      void do_r(void);
      double f(double x);
      double find_zJi(unsigned int ii);
      void do_T(void);

      void print_network(std::ostream& os, unsigned int pattern);

   public:
	   ART2Network(unsigned int inputs, unsigned int categories,
		   double epsilon, double a, double b, double c, double d, double theta, double rho);

	   ~ART2Network(void);

	   unsigned int ApplyInput(double* Inputs, unsigned int patternId);
   };

}

#endif /* !defined(_ART2_NETWORK_HPP_) */